import numpy as np
from arguments import get_args

arg = get_args()


def policy(state):
    return -np.array([0.3, 0] * arg.num2)
